package woa.binarysearch;

import java.util.Arrays;

/**
 * 找出第 k 小的距离对
 * 给定一个整数数组，返回所有数对之间的第 k 个最小距离。一对 (A, B) 的距离被定义为 A 和 B 之间的绝对差值。
 *
 * 输入：
 * nums = [1,3,1]
 * k = 1
 * 输出：0
 * 解释：
 * 所有数对如下：
 * (1,3) -> 2
 * (1,1) -> 0
 * (3,1) -> 2
 * 因此第 1 个最小距离的数对是 (1,1)，它们之间的距离为 0。
 *
 * @author wangpeng
 * @date 2021/12/24
 */
public class SmallestDistancePair {

    public static int smallestDistancePair(int[] nums, int k) {
        Arrays.sort(nums);
        int lo = 0;
        int hi = nums[nums.length - 1] - nums[0];
        while (lo < hi) {
            int mi = (lo + hi) / 2;
            int count = 0, left = 0;
            for (int right = 1; right < nums.length; right++) {
                while (nums[right] - nums[left] > mi) {
                    left++;
                }
                count += right - left;
            }
            if (count >= k) {
                hi = mi;
            } else {
                lo = mi + 1;
            }
        }
        return lo;
    }

    public static void main(String[] args) {
        System.out.println(smallestDistancePair(new int[]{1,2,3,4,5}, 5));
    }
}
